@echo off
chcp 65001 >nul
echo ========================================
echo Tiny QA Windows 安装脚本
echo ========================================
echo 正在安装 Tiny QA 知识库问答系统...

REM 检查Python版本
python --version >nul 2>&1
if errorlevel 1 (
    echo ❌ 错误: 未找到Python，请先安装Python 3.8+
    echo 下载地址: https://www.python.org/downloads/
    pause
    exit /b 1
)

echo ✅ Python已安装
python --version

REM 检查pip
pip --version >nul 2>&1
if errorlevel 1 (
    echo ❌ 错误: 未找到pip，请检查Python安装
    pause
    exit /b 1
)

echo ✅ pip已安装

REM 创建虚拟环境
echo.
echo 📦 创建虚拟环境...
if exist env (
    echo 虚拟环境已存在，跳过创建
) else (
    python -m venv env
    if errorlevel 1 (
        echo ❌ 创建虚拟环境失败
        pause
        exit /b 1
    )
    echo ✅ 虚拟环境创建成功
)

REM 激活虚拟环境
echo.
echo 🔄 激活虚拟环境...
call env\Scripts\activate.bat
if errorlevel 1 (
    echo ❌ 激活虚拟环境失败
    pause
    exit /b 1
)

REM 升级pip
echo.
echo ⬆️ 升级pip...
python -m pip install --upgrade pip

REM 安装依赖
echo.
echo 📚 安装项目依赖...
pip install -r requirements.txt
if errorlevel 1 (
    echo ❌ 安装依赖失败
    pause
    exit /b 1
)

REM 创建必要目录
echo.
echo 📁 创建必要目录...
if not exist data\database mkdir data\database
if not exist data\documents mkdir data\documents
if not exist data\vectors mkdir data\vectors
if not exist logs mkdir logs

echo ✅ 目录创建完成

REM 检查Ollama
echo.
echo 🤖 检查Ollama...
ollama --version >nul 2>&1
if errorlevel 1 (
    echo ⚠️ 警告: 未找到Ollama
    echo 请访问 https://ollama.ai 下载并安装Ollama
    echo 安装后需要下载模型: ollama pull qwen2.5:7b
) else (
    echo ✅ Ollama已安装
    ollama --version
)

echo.
echo ========================================
echo 🎉 安装完成！
echo ========================================
echo.
echo 下一步操作:
echo 1. 启动Ollama: ollama serve
echo 2. 启动Tiny QA: scripts\start.bat
echo 3. 访问: http://localhost:5002
echo.
echo 如需帮助，请查看 README.md
pause 